How to use in keyword with if python